create custom compactMap function in swift